package shuaige.jdk.多线程.base.等待和唤醒;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class 线程等待和唤醒 {
    /**
     * wait和notify用于多线程协调运行：
     * 1. 在synchronized内部可以调用wait()使线程进入等待状态；
     * 2. 必须在已获得的锁对象上调用wait()方法；
     * 3. 在synchronized内部可以调用notify()或notifyAll()唤醒其他等待线程；
     * 4. 必须在已获得的锁对象上调用notify()或notifyAll()方法；
     * 5. 已唤醒的线程还需要重新获得锁后才能继续执行。
     */
    public static void main(String[] args) throws InterruptedException {
        var taskQueue = new TaskQueue();
        var threadList = new ArrayList<Thread>();
        for (int i = 0; i < 5; i++) {
            // 执行一个线程
            var t = new Thread() {
                @Override
                public void run() {
                    // 执行task:
                    while (true) {
                        try {
                            String s = taskQueue.getTask();
                            System.out.println("execute getTask: " + s);
                        } catch (InterruptedException e) {
                            return;
                        }
                    }
                }
            };
            System.out.println("启动线程，执行：getTask()");
            t.start();
            threadList.add(t);
        }
        System.out.println("---------------getTask启动完成-----------------");
        var add = new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                // 放入task:
                String s = "t-" + Math.random();
                System.out.println("add task: " + s);
                taskQueue.addTask(s);
                try {
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    System.out.println("InterruptedException");
                }
            }
        });
        add.start();
        add.join();
        Thread.sleep(100);
        for (var t : threadList) {
            System.out.println("中断线程：" + t.getName());
            t.interrupt();
        }
    }
}

class TaskQueue {
    Queue<String> queue = new LinkedList<>();

    public synchronized void addTask(String s) {
        System.out.println("addTask 执行");
        this.queue.add(s);
        this.notifyAll();
    }

    public synchronized String getTask() throws InterruptedException {
        while (queue.isEmpty()) {
            System.out.println("getTask 等待");
            this.wait();
        }
        return queue.remove();
    }
}

